import re
import time

from common.BasePages import BasePage


class EnterprisePage(BasePage):
    """
    企业管理
    """

    def enterprise_url(self):
        self.goto_url("systemManagement/enterpriseManage")

    def create_enterprise(self):
        self.page.get_by_role("button", name="新增").click()
        self.page.get_by_role("row", name="* 企业名称 企业编码 ").get_by_placeholder("请输入企业名称").click()
        self.page.get_by_role("row", name="* 企业名称 企业编码 ").get_by_placeholder("请输入企业名称").fill(
            "张院龙企业注册")
        self.page.get_by_role("row", name="* 联系人 * 联系电话").get_by_role("cell").nth(1).click()
        self.page.get_by_role("textbox", name="请输入联系人").fill("张院龙")
        self.page.get_by_role("textbox", name="请输入联系电话").click()
        self.page.get_by_role("textbox", name="请输入联系电话").fill("15533065391")
        self.page.get_by_role("button", name="地图选择").click()
        self.page.locator(".BMap_mask").click()
        self.page.get_by_label("点击地图选取地址").get_by_role("button", name="确 定").click()
        self.page.get_by_role("button", name="确 定").click()
        self.page.get_by_text("新增成功").click()

    def edit_enterprise(self):
        self.page.get_by_role("cell", name="   ").locator("i").nth(1).click()
        self.page.get_by_role("textbox", name="请输入备注").click()
        self.page.get_by_role("textbox", name="请输入备注").fill("备注")
        self.page.get_by_role("button", name="确 定").click()

    def delete_enterprise(self):
        self.page.get_by_role("cell", name="   ").locator("i").nth(2).click()
        self.page.get_by_role("button", name="确认").click()
        self.page.get_by_text("删除成功").click()
